from django import template
from django.utils.html import format_html

register = template.Library()


@register.simple_tag
def circle_page(curr_page, loop_page):
    offset = abs(curr_page - loop_page)
    # 包含当前页向前向后的分页数量，设置为3 则当前页最多是向前第三页向后第三页  （12（3）45）
    if offset < 3:
        if curr_page == loop_page:
            page_ele = f'<b>{loop_page}</b>'
        else:
            page_ele = f'<a href="?page={loop_page}">{loop_page}</a>'

        return format_html(page_ele)
    else:
        return ''
